import java.util.HashMap;
import java.util.Map;

/**
 * 560. 和为 K 的子数组
 * https://leetcode.cn/problems/subarray-sum-equals-k/description/
 */
class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0, ret = 0;
        for(int x : nums) {
            sum += x;//计算当前前缀和
            ret += map.getOrDefault(sum - k, 0);//统计结果
            map.put(sum, map.getOrDefault(sum, 0) + 1);//把当前数组丢到哈希表里
        }
        return ret;
    }

    /**
     *974. 和可被 K 整除的子数组
     * https://leetcode.cn/problems/subarray-sums-divisible-by-k/description/
     */
    public int subarraysDivByK(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0 % k, 1);
        int sum = 0, ret = 0;
        for(int x : nums) {
            sum += x;//计算此时的前缀和
            int r = (sum % k + k) % k;
            ret += map.getOrDefault(r, 0);//统计结果
            map.put(r, map.getOrDefault(r, 0) + 1);//把使用完的余数丢到Hash表里
        }
        return ret;
    }
}